import torch
from torch.utils.data import Dataset

# to do 
class weatherDataset(Dataset):
    def __init__(self, path, partition='train'):
        pass

    def __getitem__(self, item):
        data = torch.randn(24,1689, 2520)
        label = torch.randn(60,1689, 2520)
        return data, label

    def __len__(self):
        return 1
